package cn.com.guage.dtm.leetcode.hot100;

import java.util.*;
import java.util.stream.Collectors;

/**
 * 给你一个整数数组 nums ，判断是否存在三元组 [nums[i], nums[j], nums[k]]
 * 满足 i != j、i != k 且 j != k ，同时还满足 nums[i] + nums[j] + nums[k] == 0 。请你返回所有和为 0 且不重复的三元组。
 * <p>
 * 注意：答案中不可以包含重复的三元组。
 */
public class LeetCode_15_三数之和 {
    public static void main(String[] args) {
        int[] nums = {-1, 0, 1, 2, -1, -4};
        int[] nums2 = {0,0,0,0};
        List<List<Integer>> list = threeSum(nums2);
        System.out.println(list);
    }

    /**
     * 1. 用暴力破解法，性能较差
     * 2. 309 / 314 个通过的测试用例
     * 3. 使用res.stream().distinct()对元素去重
     * @param nums
     * @return
     */
    public static List<List<Integer>> threeSum(int[] nums) {
        Arrays.sort(nums);
        List<List<Integer>> res = new ArrayList<>();
        for(int i=0; i<nums.length-2; i++) {
            for(int j=i+1; j<nums.length-1; j++) {
                int sum = nums[i] + nums[j];
                for(int k=j+1; k<nums.length; k++) {
                    List<Integer> triple = new ArrayList<>();
                    if (sum+nums[k] == 0){
                        triple.add(nums[i]);
                        triple.add(nums[j]);
                        triple.add(nums[k]);
                    }

                    if (triple.size() > 0){
                        Collections.sort(triple);
                        res.add(triple);
                        break;

                    }
                }
            }
        }
        //删除重复的三元组
        List<List<Integer>>  result = res.stream().distinct().collect(Collectors.toList());
        return result;
    }
    /**
     * 1. 用暴力破解法，性能较差
     * 2. 309 / 314 个通过的测试用例
     * 3. 使用 Set<List<Integer>>对元素去重
     * @param nums
     * @return
     */
    public static List<List<Integer>> threeSum2(int[] nums) {
        Arrays.sort(nums);
        List<List<Integer>> res = new ArrayList<>();
        for(int i=0; i<nums.length-2; i++) {
            for(int j=i+1; j<nums.length-1; j++) {
                int sum = nums[i] + nums[j];
                for(int k=j+1; k<nums.length; k++) {
                    List<Integer> triple = new ArrayList<>();
                    if (sum+nums[k] == 0){
                        triple.add(nums[i]);
                        triple.add(nums[j]);
                        triple.add(nums[k]);
                    }

                    if (triple.size() > 0){
                        Collections.sort(triple);
                        res.add(triple);
                        break;

                    }
                }
            }
        }
        //使用Set集合进行去重
        Set<List<Integer>> set = new HashSet<List<Integer>>();
        set.addAll(res);
        //删除重复的三元组
        return new ArrayList(set);
    }
}
